#
# Copyright 2005, 2006,2010 Zuza Software Foundation
#
# This file is part of the translate-toolkit
#
# translate is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# translate is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, see <http://www.gnu.org/licenses/>.

"""
Compile XLIFF and Gettext PO localization files into Gettext MO (Machine Object) files.

See: http://docs.translatehouse.org/projects/translate-toolkit/en/latest/commands/pocompile.html
for examples and usage instructions.
"""

from translate.convert import convert
from translate.misc.multistring import multistring
from translate.storage import factory, mo


def _do_msgidcomment(string):
    return f"_: {string}\n"


class POCompile:
    @staticmethod
    def convertstore(inputfile, includefuzzy=False):
        outputfile = mo.mofile()
        for unit in inputfile.units:
            if (
                unit.istranslated()
                or (unit.isfuzzy() and includefuzzy and unit.target)
                or unit.isheader()
            ):
                mounit = mo.mounit()
                if unit.isheader():
                    mounit.source = ""
                else:
                    mounit.source = unit.source
                    context = unit.getcontext()
                    if unit.msgidcomment:
                        if mounit.hasplural():
                            mounit.source = multistring(
                                _do_msgidcomment(unit.msgidcomment) + mounit.source,
                                *mounit.source.strings[1:],
                            )
                        else:
                            mounit.source = (
                                _do_msgidcomment(unit.msgidcomment) + mounit.source
                            )
                    elif context:
                        mounit.msgctxt = [context]
                mounit.target = unit.target
                outputfile.addunit(mounit)
        return bytes(outputfile)


def convertmo(inputfile, outputfile, templatefile, includefuzzy=False):
    """Reads in a base class derived inputfile, converts using pocompile, writes to outputfile."""
    # note that templatefile is not used, but it is required by the converter...
    inputstore = factory.getobject(inputfile)
    if inputstore.isempty():
        return 0
    convertor = POCompile()
    outputmo = convertor.convertstore(inputstore, includefuzzy)
    # We have to make sure that we write the files in binary mode, therefore we
    # reopen the file accordingly
    outputfile.close()
    with open(outputfile.name, "wb") as fh:
        fh.write(outputmo)
    return 1


def main():
    formats = {
        "po": ("mo", convertmo),
        "xlf": ("mo", convertmo),
        "xliff": ("mo", convertmo),
    }
    parser = convert.ConvertOptionParser(formats, usepots=False, description=__doc__)
    parser.add_fuzzy_option()
    parser.run()


if __name__ == "__main__":
    main()
